# -*- coding: utf-8 -*-
"""
Created on Sun Mar 12 10:20:31 2023

@author: forla
"""

import tensorflow as tf
from tensorflow.keras.layers import Dense, Dropout

from model.mpcnn_model import MPCNN_Model

class MPCNoEmbedModel(MPCNN_Model):
    def __init__(self, input_shape: int = 9, num_classes: int = 27, *args, **kwargs):
        super().__init__(input_shape=input_shape, num_classes=num_classes, *args, **kwargs)

    def create_model_io(self, input_shape: int=9, num_classes: int=27):
        inp = tf.keras.Input(input_shape)
        out = Dense(50, activation="relu", use_bias=False, name="layer1")(inp)
   #     out = Dense(50, activation="relu", use_bias=False, name="layer2")(out)
  #      out = Dense(50, activation="relu", use_bias=False, name="layer4")(out)
        out = Dense(num_classes, activation="softmax",use_bias=False, name="layer3")(out)
        return inp, out
    
class MPCNoEmbedModel_2L(MPCNN_Model):
    def __init__(self, input_shape: int = 7, num_classes: int = 7, *args, **kwargs):
        super().__init__(input_shape=input_shape, num_classes=num_classes, *args, **kwargs)

    def create_model_io(self, input_shape: int=7, num_classes: int=7):
        inp = tf.keras.Input(input_shape)
        out = Dense(10, activation="relu", use_bias=False, name="layer1")(inp)
   #     out = Dense(50, activation="relu", use_bias=False, name="layer2")(out)
  #      out = Dense(50, activation="relu", use_bias=False, name="layer4")(out)
        out = Dense(num_classes, activation="softmax",use_bias=False, name="layer3")(out)
        return inp, out